package socket;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;

/**
 * 聊天室的服务端 版本4
 * 这个版本我们实现将消息群发给所有客户端
 *
 * 思路:
 * 当某一个客户端发送过来消息后,我们需要使用所有对客户端的输出流将消息发给所有客户端
 * 这里的问题是:
 * 每个线程启动后负责与一个客户端交互(线程任务是一个ClientHandler实例).而每个线程都只拥有自己客户端的
 * 输出流,可以给自己对应的客户端发送消息.
 * 只有让每个线程都能访问到其它线程对应的输出流,才能将消息转发给其它客户端.
 *
 * 解决办法:
 * 核心思想:多个ClientHandler之间想互访对方的输出流时,只需要将它们各自的输出流存放在一个大家都能看得到
 * 的地方即可.
 *
 * 由于ClientHandler是Server的内部类,而内部类可以访问外部类的成员属性.
 * 因此我们在Server上定义一个集合,所有的线程任务(ClientHandler)都将自己对应的客户端输出流存入该集合
 * 这样一来,每个ClientHandler都可以通过遍历集合得到所有客户端的输出流,从而给每个客户端发送消息了
 *
 * 实现:
 * 1:在Server上定义一个属性List<PrintWriter> allOut
 * 2:每个ClientHandler在获取对应客户端的输出流后都存入到这个集合中
 * 3:每当读取到来自客户端发送过来的消息后就遍历allOut集合,将消息陆续发送给所有客户端
 *   达到广播消息的目的
 */

/** 聊天室的服务端 */
public class Server4 {
    /*
    java.net.ServerSocket
    运行在服务端的ServerSocket的主要任务：
    1.向服务端申请服务端口，客户端就是通过该端口与服务器建立连接的
    2.监听服务端口，一旦一个客户端连接了就会立即创建一个Socket实例与之交互
      我们将ServerSocket比喻为某客服中心的"总机"
     */
    private ServerSocket serverSocket;

    //V4:该集合用于保存所有客户端的输出流，用于广播消息
    private List<PrintWriter> allOut = new ArrayList<>();

    public Server4(){
        try {
            /*
            如果端口被其他程序占用了，则会发生java.net.BindException异常：Address already in use解决办法：
            1.更换端口号
            2.杀死占用该端口的进程(通常发生在服务端启动两次导致的情况下)
             */
            System.out.println("正在启动服务端");
            serverSocket = new ServerSocket(8088);
            System.out.println("服务端启动完毕!");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public void start(){
        /*
        ServerSocket提供的重要方法
        Socket accept()
        用于接收客户端的连接，并返回一个Socket实例与连接的客户端交互
        该方法是一个阻塞方法，调用后程序会"卡住"，直到一个客户端连接为止
         */
        try {
            while (true){
                System.out.println("等待客户端连接");
                Socket socket = serverSocket.accept(); //接电话
                System.out.println("一个客户端连接上了");
                //启动一个线程负责与该客户端进行交互
                ClientHandler handler = new ClientHandler(socket);
                Thread t = new Thread(handler);
                t.start();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        Server4 server = new Server4();
        server.start();
    }

    /** 该线程负责与指定的客户端交互 */
    private class ClientHandler implements Runnable{
        //通过设置属性以及构造器传参的方法解决socket报错的问题
        private Socket socket;
        private String host; //记录该客户端的IP地址

        public ClientHandler(Socket socket){
            this.socket = socket;
            host = socket.getInetAddress().getHostAddress(); //通过socket获取远端计算机的IP地址的字符串格式
        }

        public void run(){
            try {
                InputStream in = socket.getInputStream(); //低级流
                InputStreamReader isr = new InputStreamReader(in, StandardCharsets.UTF_8);
                BufferedReader br = new BufferedReader(isr);

                //V3：通过socket获取输出流用于给客户端发送消息
                OutputStream out = socket.getOutputStream();
                OutputStreamWriter osw = new OutputStreamWriter(out,StandardCharsets.UTF_8);
                BufferedWriter bw = new BufferedWriter(osw);
                PrintWriter pw = new PrintWriter(bw,true);

                //V4：将对应该客户daunt的输入流存入到共享集合allOut中，便于其它线程可以访问到
                allOut.add(pw);

                /*
                当我们通过br.readLine()读取来自远端计算机(客户端)发送过来的字符串时
                远端计算机只有调用socket.close()正常与我们断开连接时，readLine()才会返回null
                注意：如果客户端异常中断(强制杀死进程)，此时readLine()会抛出异常
                 */
                String message;
                while((message=br.readLine())!=null){
                    System.out.println(host+"说："+message);

                    //V3:将读取到的消息回复给客户端
                    //pw.println(host+"说："+message);

                    //V4:遍历allOut集合，给所有客户端发送这个消息
                    for(PrintWriter o : allOut){
                        o.println(host+"说："+message);
                    }
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}
